/*
 * @(#)MyGraphTreeModel.java   09/09/13
 *
 * Copyright (C) 2009 Ecole des Mines d'Ales
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 * Contact: yulinz88@gmail.com
 *
 *
 *
 */



package fr.ema.arch3d.gea.config;

import org.jgraph.event.GraphModelEvent;
import org.jgraph.event.GraphModelListener;
import org.jgraph.graph.GraphModel;

import java.util.Enumeration;
import java.util.Vector;

import javax.swing.tree.DefaultTreeModel;
import javax.swing.tree.TreeNode;

/**
 * Class description
 *
 *
 * @version        1.0, 09/09/13
 * @author         Huaxi (Yulin) ZHANG
 */
public class MyGraphTreeModel extends DefaultTreeModel implements GraphModelListener {

    /**
     * Look how we wrapp the GraphModel into a GraphModelTreeNode to construct
     * the GraphTreeModel and satisfy both the GraphModel and the TreeModel interfaces.
     * @param model
     */
    public MyGraphTreeModel(GraphModel model) {
        super(new GraphModelTreeNode(model));
    }

    public void graphChanged(GraphModelEvent e) {
        reload();
    }

    /**
     * Class description
     *
     *
     * @version        1.0, 09/09/13
     * @author         Huaxi (Yulin) ZHANG
     */
    public static class GraphModelTreeNode implements TreeNode {

        /** Field description */
        protected GraphModel model;

        /**
         * Constructs ...
         *
         *
         * @param model
         */
        public GraphModelTreeNode(GraphModel model) {
            this.model = model;
        }

        public Enumeration children() {
            Vector v = new Vector();

            for (int i = 0; i < model.getRootCount(); i++) {
                v.add(model.getRootAt(i));
            }

            return v.elements();
        }

        public boolean getAllowsChildren() {
            return true;
        }

        public TreeNode getChildAt(int childIndex) {
            return (TreeNode) model.getRootAt(childIndex);
        }

        public int getChildCount() {
            return model.getRootCount();
        }

        public int getIndex(TreeNode node) {
            return model.getIndexOfRoot(node);
        }

        public TreeNode getParent() {
            return null;
        }

        public boolean isLeaf() {
            return false;
        }

        public String toString() {
            return model.toString();
        }
    }
}
